package com.csw.scala

import scala.io.Source

object Demo20ScalaWordCount {
  def main(args: Array[String]): Unit = {
    //1、读取文件
    val list: List[String] = Source.fromFile("Scala/data/words.txt").getLines().toList
    //2、将每一行中的多个单词拆分出来
    val words: List[String] = list.flatMap((f: String) => f.split(","))
    //3、对单词进行分组，将同一个单词分到同一个组中
    val groups: Map[String, List[String]] = words.groupBy(k => k)
    //4、统计单词的数量
    val wordCount =groups.map(kv =>{
      //一个单词
      val word: String = kv._1
      //当前的一个单词构建的集合
      val size: List[String] = kv._2
      //统计单词的数量
      val count: Int = size.length
      //返回单词和单词的数量
      (word,count)
    })
    /**
      * foreach:  遍历集合，将集合中的元素一个一个的传递给后面的函数，没有返回值
      *
      */
    wordCount.foreach(kv => println(kv))

    /**
      * 简写
      */
    //链式调用
    Source
      .fromFile("Scala/data/words.txt")//读取文件
      .getLines()//获取所有行
      .toList//转换成list集合
      .flatMap((i: String) => i.split(","))//将单词拆分出来
      .groupBy(word => word)//按照单词进行分组
      .map(kv => (kv._1,kv._2.length))//统计单词的数量
      .foreach(kv => println(kv))//打印结果
  }

}
